import os
import sys
from pathlib import Path

# 将项目根目录添加到 Python 路径
ROOT_DIR = Path(__file__).parent.parent
sys.path.append(str(ROOT_DIR))

from app import create_app, db
from app.models.user import User

def create_admin_user():
    app = create_app()
    with app.app_context():
        # 检查是否已存在管理员用户
        admin = User.query.filter_by(username='admin').first()
        if not admin:
            # 创建管理员用户
            admin = User(
                username='admin',
                email='admin@example.com',
                is_admin=True
            )
            admin.set_password('admin123')
            db.session.add(admin)
            db.session.commit()
            print("管理员用户创建成功")
        else:
            # 如果已存在，确保其具有管理员权限
            if not admin.is_admin:
                admin.is_admin = True
                db.session.commit()
                print("已更新管理员权限")
            print("管理员用户已存在")

if __name__ == '__main__':
    create_admin_user() 